/*
** server.c -- a stream socket server demo
*/

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <sys/wait.h>
#include <signal.h>

#include "resposta.h"
#include "tempo.h"

#define PORT "39410"  // the port users will be connecting to
#define BACKLOG 10	 // how many pending connections queue will hold
#define MAXDATASIZE 100




/* --------------------------------------------------------------------------*/
/**
 * @Synopsis  
 * Função auxiliar para controlar os processos filhos
 *
 * @Param s - flag 
 */
/* ----------------------------------------------------------------------------*/
void sigchld_handler(int s)
{
	while(waitpid(-1, NULL, WNOHANG) > 0);
}

// get sockaddr, IPv4 or IPv6
/* --------------------------------------------------------------------------*/
/**
 * @Synopsis  
 * Função que obtem o endereço do socket
 *
 * @Param sa - estrutura de dados do tipo sockaddr
 *
 * @Returns   endereço do socket (IPv4 ou IPv6)
 */
/* ----------------------------------------------------------------------------*/
void *get_in_addr(struct sockaddr *sa)
{
	if (sa->sa_family == AF_INET) {
		return &(((struct sockaddr_in*)sa)->sin_addr);
	}

	return &(((struct sockaddr_in6*)sa)->sin6_addr);
}

/* --------------------------------------------------------------------------*/
/**
 * @Synopsis  
 * Função que controla o fluxo de execução do servidor, recebendo os parâmetros
 * passados pelo cliente, montando uma resposta de acordo com esses parâmetros,
 * e devolvendo essa resposta ao cliente
 *
 * @Returns  inteiro indicando se o fluxo de execução do programa foi bem
 * sucedido 
 */
/* ----------------------------------------------------------------------------*/
int main(void)
{
	int sockfd, new_fd;  // listen on sock_fd, new connection on new_fd
	struct addrinfo hints, *servinfo, *p;
	struct sockaddr_storage their_addr; // connector's address information
	socklen_t sin_size;
	struct sigaction sa;
	int yes=1;
	char s[INET6_ADDRSTRLEN];
	int rv;
/*Novidades******************************************/
	int numbytes;
	char buf[MAXDATASIZE];
	char *resposta;


    struct timeval startTime; //Estrutura contendo o tempo de início de execução
    struct timeval endTime; //Estrutura contendo o tempo final de execução
    double time;
    /*******************************************/	

	memset(&hints, 0, sizeof hints);
	hints.ai_family = AF_UNSPEC;
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_flags = AI_PASSIVE; // use my IP

	if ((rv = getaddrinfo(NULL, PORT, &hints, &servinfo)) != 0) {
		fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(rv));
		return 1;
	}

	// loop through all the results and bind to the first we can
	for(p = servinfo; p != NULL; p = p->ai_next) {
		if ((sockfd = socket(p->ai_family, p->ai_socktype,
				p->ai_protocol)) == -1) {
			perror("server: socket");
			continue;
		}

		if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &yes,
				sizeof(int)) == -1) {
			perror("setsockopt");
			exit(1);
		}

		if (bind(sockfd, p->ai_addr, p->ai_addrlen) == -1) {
			close(sockfd);
			perror("server: bind");
			continue;
		}

		break;
	}

	if (p == NULL)  {
		fprintf(stderr, "server: failed to bind\n");
		return 2;
	}

	freeaddrinfo(servinfo); // all done with this structure

	if (listen(sockfd, BACKLOG) == -1) {
		perror("listen");
		exit(1);
	}

	sa.sa_handler = sigchld_handler; // reap all dead processes
	sigemptyset(&sa.sa_mask);
	sa.sa_flags = SA_RESTART;
	if (sigaction(SIGCHLD, &sa, NULL) == -1) {
		perror("sigaction");
		exit(1);
	}

	printf("server: waiting for connections...\n");

	while(1) {  // main accept() loop
		sin_size = sizeof their_addr;
		new_fd = accept(sockfd, (struct sockaddr *)&their_addr, &sin_size);
		if (new_fd == -1) {
			perror("accept");
			continue;
		}

		inet_ntop(their_addr.ss_family,
			get_in_addr((struct sockaddr *)&their_addr),
			s, sizeof s);
		printf("server: got connection from %s\n", s);

		if (!fork()) { // this is the child process
			close(sockfd); // child doesn't need the listener

/*receiving request******************************************/			
			if ((numbytes = recv(new_fd, buf, MAXDATASIZE-1, 0)) == -1) {
	           perror("recv");
	           exit(1);
            }
			
            //Inicia a contagem de tempo, antes do write (recv)
            gettimeofday(&startTime, NULL);

	        buf[numbytes] = '\0';
			resposta = preparaResposta(buf);
			
			//Termina a contagem de tempo, após o read (send)
            gettimeofday(&endTime, NULL);

			if (send(new_fd, resposta, strlen(resposta), 0) == -1)
				perror("send");
			
            time = calculaTempo(startTime, endTime);
            printf("Processing Time: %.2lf microseconds\n", time);
			
            close(new_fd);
			exit(0);
        }
       close(new_fd);  // parent doesn't need this
    }
	return 0;
}

